package com.chenyun.tenant;

import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.chenyun.common.vo.ApplicationProperties;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;

import java.util.ArrayList;
import java.util.List;

/**
 * 多租户
 *
 * @author: Lonni
 * @date: 2023/12/11 0011 17:16
 */
public class TenantHandlerImpl implements TenantLineHandler {


    volatile static Boolean isInit = false;

    /**
     * 多租户标识
     */
    private static final String SYSTEM_TENANT_ID = "tenant_id";


    /**
     * 需要过滤的表
     */
    private static final List<String> IGNORE_TENANT_TABLES = new ArrayList<>();


    public TenantHandlerImpl(ApplicationProperties properties) {
        if (!isInit) {
            if (CollectionUtil.isNotEmpty(properties.getTenantIgnoreTables())) {
                IGNORE_TENANT_TABLES.addAll(properties.getTenantIgnoreTables());
            }
            isInit = true;
        }
    }


    /**
     * 获取租户 ID 值表达式，只支持单个 ID 值
     * <p>
     *
     * @return 租户 ID 值表达式
     */
    @Override
    public Expression getTenantId() {
        return new LongValue(TenantContext.me.tenantId());
    }

    /**
     * 获取租户字段名
     * <p>
     * 默认字段名叫: tenant_id
     *
     * @return 租户字段名
     */
    @Override
    public String getTenantIdColumn() {
        return SYSTEM_TENANT_ID;
    }

    /**
     * 根据表名判断是否忽略拼接多租户条件
     * <p>
     * 默认都要进行解析并拼接多租户条件
     *
     * @param tableName 表名
     * @return 是否忽略, true:表示忽略，false:需要解析并拼接多租户条件
     */
    @Override
    public boolean ignoreTable(String tableName) {
        return IGNORE_TENANT_TABLES.contains(tableName)
                || TenantContext.me.ignore();
    }
}



